package config

import (
	"fmt"
	"github.com/spf13/viper"
)

// 初始化viper
func NewViper(path string) (*viper.Viper, error) {
	v := viper.New()
	// 添加路径
	v.AddConfigPath(".")
	// 设置文件类型
	v.SetConfigType("yml")
	// 添加配置文件
	v.SetConfigFile(path)
	// 读取配置文件
	if err := v.ReadInConfig(); err != nil {
		fmt.Printf("读取配置文件失败 -> %s\n", v.ConfigFileUsed())
		return v, err
	}
	return v, nil
}
